Distributionally Robust Mixture-of-Experts Training
分布鲁棒的混合专家训练
🧭 MoE训练 · 分布鲁棒优化 ★ 深度解读
New York University
提出DRMoET,将MoE专家作为内生DRO组,动态加权高损失路由结果,提升路由鲁棒性和下游性能
📌 工作参考价值(4 点)
- 在电商/广告的大规模MoE推荐模型中,可在现有负载均衡之外引入DRMoET式的专家级损失加权:用EMA平滑的激活加权损失动态调整每个专家的训练权重,无需修改前向路由,即可提升非热门专家在流量波动或分布偏移时的表现,降低线上指标抖动。
- 将「内生组DRO」思想迁移到多任务学习或用户分群:不依赖预定义分组,而是根据模型当前的路由/分配结果动态识别高损失子群,进行针对性加权,比固定分组更适应数据分布变化。
- 工程实现上,该方法只需维护每层专家数的EMA状态和权重(O(LE)标量),计算开销可忽略,且权重detach避免干扰主损失;可低成本部署到现有训练框架,作为对load-balancing的补充。
- 用forced misrouting probe(强制选择中位专家)评估模型对非最优路由的鲁棒性,可作为线上路由策略变更前的离线诊断工具。